#!/usr/bin/env python
# -*- coding: utf-8 -*-
#===============================================================================
# Copyright 2011 zod.yslin
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# 
# Author: zod.yslin
# Email: 
# File Name: execute.py
# Description: 
#   1. SELECT
#   2. INSERT
#   3. DELETE
# Edit History: 
#   2011-12-05    File created.
#===============================================================================

# 引入 MySQL 模組
import MySQLdb
_host = "192.168.xxxx"
_user = "xxxxxx"
_password = "xxxxxxxxx"
_database = "xxxxxxxx"
_table = "malware_test"

def select(cursor):
    # 執行 SQL 語句
    cursor.execute("SELECT * FROM " + _table)
    result = cursor.fetchall()

    # 輸出結果
    for record in result:
        print record

def insert(cursor, assetName, pkgName, virusName, type, updateTime):
    """
    assetName = 'test porn'
    pkgName = 'com.test.porn'
    virusName = 'TESTPorn'
    type = 'Porn'
    updateTime = '2011-04-12 10:12:23'
    """
    cursor.execute("""INSERT INTO """ + _table +""" (assetName, pkgName, virusName, type, updateTime) VALUES 
                   (%s, %s, %s, %s, %s)""", 
                   (assetName, pkgName, virusName, type, updateTime))

def delete(cursor, pkgName):
    cursor.execute("""DELETE FROM """ + _table +""" WHERE pkgName = %s""", (pkgName))


def main():
    # 連接到 MySQL
    db = MySQLdb.connect(host=_host, user=_user, passwd=_password, db=_database, charset="utf8") #这里要注明是utf8字符集，文件开头最好也加上utf-8的声明
    cursor = db.cursor()

    select(cursor)
    insert(cursor, 'IamPorn', 'com.test.porn', 'TestPorn', 'Porn', '2011-04-12 11:12:20')
    delete(cursor, 'com.test.porn')
    delete(cursor, 'com.test.porn1')
    select(cursor)

     

if __name__=='__main__':
    main()
